"""deepseek官网接口文档：https://openai.deepseek.cn/docs"""

from openai import OpenAI

def deepseek_predict(prompt: str):
    client = OpenAI(api_key="sk-e15aa1537f844ae1bdeb7dd5cf726e70", base_url="https://api.deepseek.com")

    response = client.chat.completions.create(
        model="deepseek-chat",
        messages=[
            {"role": "system", "content": "You are a helpful assistant"},
            {"role": "user", "content": prompt},
        ],
        stream=False
    )
    return response.choices[0].message.content




if __name__ == '__main__':
    context = "请详细介绍一下deepseek模型的训练策略、模型的架构、模型的参数、模型的训练数据、模型的评估指标"
    print(deepseek_predict(context))
